fixed typo
#6 simplified assertion