Fixed, additionally the test cases now make sure that a correctly shaped sub-sample is passed.