Added the needed test