Updating tests fro carelessly merged PR