Fixes missing return argument