fix for return types