Added assertion for source as well to prevent regression