Fix tests

http://jira.webinos.org/browse/WP-1220