change break to return