fix entropy plot test