fix bug: in test
fix: prob-14