Fixed a bug with initialization
Fixed a bug with initialize