fixes sample test to pass