fix tests

Signed-off-by: Stephan Pavlovic <stephan@railslove.com>