fix return value

Quick and Dirty fix