add return

@csnauwaert