Fix return from behavior