Need to include print_function to make tests pass