Addressed note on test