Adding assertion for tests