Fix IMDB test