Add shorter test