Update tests accordingly