Fixed bug with return type