Add back method that was needed for tests