Incorrect initialization in env_stocktrading for single asset

In finrl/meta/env_stock_trading/env_stocktrading.py in function _initiate_state when the environment initializated for single stock - it always initialize as zero.

From line 419
else:
# for single stock
state = (
[self.initial_amount]
+ [self.data.close]
+ [0] * self.stock_dim
+ sum(([self.data[tech]] for tech in self.tech_indicator_list), [])
)
I think it should be something like that:
else:
# for single stock
state = (
[self.initial_amount]
+ [self.data.close]
+ [self.num_stock_shares[0]] * self.stock_dim
+ sum(([self.data[tech]] for tech in self.tech_indicator_list), [])
)